import os
import rasterio
from rasterio.transform import from_origin

# 输入和输出文件夹路径
input_path = './mean_npp/'  # 原始 .tiff 文件夹路径
output_path = './processed_npp_frac_10/'  # 处理后的 .tiff 文件夹路径

# 确保输出文件夹存在
os.makedirs(output_path, exist_ok=True)

# 遍历输入文件夹中的所有 .tiff 文件
for file in sorted(os.listdir(input_path)):
    if file.endswith('.tiff') or file.endswith('.tif'):  # 只处理 .tiff 文件
        input_file = os.path.join(input_path, file)
        output_file = os.path.join(output_path, file)

        # 打开 .tiff 文件
        with rasterio.open(input_file) as src:
            # 读取元数据
            nodata = src.nodata
            meta = src.meta
            image = src.read(1)  # 读取第一个波段
            image_except_nodata = image[image!=nodata]#*src.scales
            # 对像素值除以 10
            processed_data = image / 10

            # 更新元数据
            meta.update(dtype=processed_data.dtype)

            # 保存处理后的数据到输出文件夹
            with rasterio.open(output_file, 'w', **meta) as dst:
                dst.write(processed_data, 1)

        print(f"处理完成: {file} -> {output_file}")